How to restrict the choice of a user in python